package cn.tedu.v4;

import java.io.*;
import java.net.*;
import java.util.*;

/**
 * Version 4 多人聊天室  ---消息转发【广播】功能
 * 建议：单线聊天，多人聊天，多人聊天+QQ模式  【消息转发功能】
 * 拓展功能：建议增加第三个版本 Version 3
 *
 * 消息转发步骤：服务器给所有的客户端写出内容，客户端之间相互收到内容
 * 1.集合存储所有写出的客户端--服务器端写出给所有客户端
 * 2.广播消息给所有客户端----broadcastMessage 广播留言
 *   注意：写在ClientHandler上，必须同步方法
 * 3.将客户端的输出流添加 allOut集合中【必须同步块】
 *   注意：写在ClientHandler的构造器--所有客户端具备功能：消息转发
 * 4.广播消息给所有客户端
 *   注意：写在ClientHandler的run--客户端写出的消息，读入方法转发出去
 * 5.客户端退出，从  allOut 集合中移除
 *   注意：写在ClientHandler的run--finally【一定移除链接】
 */
public class Server {
    private ServerSocket serverSocket;
    private static final  Integer PORT=4545;
    //1.私有属性：集合存储一个客户端链接，发送文本，转发给其他用户
    private List<PrintWriter> allOut=new ArrayList<>();

    public Server() {
        try {
            serverSocket=new ServerSocket(PORT);
        } catch (BindException e){
            System.out.println("端口号重复申请，建议换个端口号");
        } catch (IOException e) {
            System.out.println("申请端口号失败....");
        }
    }
    public static void main(String[] args) {
        new Server().start();
    }


    public void start(){
        try {
            while(true) {
                System.out.println("----等待客户端链接-----");
                Socket socket = serverSocket.accept();
                String ip = socket.getInetAddress().getHostAddress();
                System.out.println("李佳怡" + ip + "连接成功");

                new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("服务器端通讯断开了");
        } finally {
            if(serverSocket!=null&&!serverSocket.isClosed()){
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
            }
        }
    }


    //2.广播消息给所有的客户端---功能 broadcastMessage 广播留言
    // 2.1 broadcastMessage功能：提供转发给客户端消息【加同步锁】
    //     理解：来一个客户端，只允许当前客户端转发消息
    // 2.2 String message 参数：转发消息---代表:消息
    private synchronized void broadcastMessage(String message){
        //2.3 增强for:  allOut 存储所有客户端
        // 【理解：来很多客户端，很多的客户端遍历出来】
        for(PrintWriter pw: allOut){ //allOut保存所有链接的客户端
            pw.println("消息广播："+message); //理解：消息转发出去
        }
    }

    private class ClientHandler implements Runnable{
        private Socket socket;
        private String ip;
        private BufferedReader br;
        private PrintWriter pw;

        public ClientHandler(Socket s){
            this.socket=s;
            ip=socket.getInetAddress().getHostAddress();
            try {
                br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
                pw=new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(),"UTF8")),true);
                //3.所有客户端具备消息转发
                // 理解：客户端写出文字，服务器负责转发文字
                //监视同步对象：allOut---存所有客户端--
                //客户端存在线程中---执行存在执行顺序不一致情况
                synchronized (allOut) {
                    allOut.add(pw);  //客户端写出文字，存集合中
                }

            } catch (IOException e) {
                System.out.println("客户端断开通讯~~");
            }
        }

        public void run() {
            try {
                new Thread(()->{
                    try {
                        String message;
                        while((message=br.readLine())!=null){
                            System.out.println("客户端说："+message);
                            System.out.println("服务器请输入：");
                            //4.广播消息给所有的客户端
                            //broadcastMessage转发：写出文本内容
                            //message 客户端发给服务器的内容
                            //把当前客户端写出的内容，通过服务器读入：message
                            // 转发给所有的客户端
                            broadcastMessage(message);
                        }
                    } catch (IOException e) {
                        System.out.println("客户端读入失败！");
                    }
                }).start();
                Scanner input =new Scanner(System.in);
                while (true) {
                    System.out.println("服务器请输入：");
                    String say=input.nextLine();
                    pw.println("服务器说："+say);
                }
            } catch (NullPointerException e){
                System.out.println("IO流传入对象，" +
                        "有对象为null！立马核对，" +
                        "建议：查看try块中对象可能赋值为null");
            } finally {
                //5.一旦有 客户端断开，客户端从 allOut 集合中移除 输出流
                // 必须同步块，结束客户端不一致
                synchronized (allOut) {  //监视allOut集合中客户端有没有断开
                    allOut.remove(pw);   //allOut集合客户端断开链接，移除写出流
                }
                if(br!=null)
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("结束通讯：关闭失败！");
                    }
                if(pw!=null) pw.close();
                if(socket!=null&&!socket.isClosed())
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败~");
                    }
            }
        }
    }
}
